# 基本元素操作

from __future__ import print_function
import torch

# 无初始化5行3列矩阵
x = torch.empty(5, 3)
print(x)

# 有初始化5行3列矩阵
# rand是均匀分布，randn是正态分布，randi（max）是0-max上均匀分布的整数
x = torch.rand(5, 3)
print(x)

# 全零矩阵并指定数据元素类型为long
x = torch.zeros(5, 3, dtype=torch.long)
print(x)

# 直接传入数据
x = torch.tensor([2.5, 3.3])
print(x)

# 通过已有张量创建相同尺寸的新张量
# 全是1
x = x.new_ones(5, 3, dtype=torch.double)
print(x)
# 随机赋值
y = torch.randn_like(x, dtype=torch.float)
print(y)

# 获取张量的尺寸
print(x.size())
# size为元组
a, b = x.size()
print("a:", a)
print("b:", b)
